package cn.heiliu.selenium.config;

import cn.hutool.core.util.StrUtil;
import lombok.Builder;
import lombok.Getter;

import java.io.File;
import java.util.UUID;

/**
 * 浏览器启动配置
 * @author wangyelou
 * @date 2021/6/14
 */
@Builder
@Getter
public class SetupConfig {

    /**
     * 池大小
     */
    @Builder.Default
    private int poolSize = 10;

    /**
     * 最大连续失败次数
     */
    @Builder.Default
    private int maxFailedNum = 10;

    /**
     * 浏览器执行路径
     */
    private String binPath;

    /**
     * 引擎位置
     */
    private String enginePath;

    /**
     * 是否加载图片
     */
    private boolean loadImage = false;

    /**
     * 是否无头
     */
    private boolean headless = false;

    /**
     * 文件下载位置
     */
    private String downDir;

    /**
     * 代理认证插件位置
     */
    private String proxyAuthExt;

    /**
     * 获取下载目录
     * @param uid
     * @return
     */
    public String getDownDir(String uid) {
        return StrUtil.hasEmpty(downDir) ? uid : StrUtil.removeSuffix(downDir, File.separator) + File.separator + uid;
    }

}
